# example6_19_violin.py
# coding=utf-8
import numpy as np
import matplotlib.pyplot as plt

# 读取葡萄酒数据
# 第2列表示苹果酸（Malic acid）的含量
# 第5列表示镁（Magnesium）的含量
# 第13列表示脯氨酸（Proline）的含量
data = np.loadtxt("wine.data", delimiter=",", usecols=(1, 4, 12))
# print(data)
plt.rcParams["font.sans-serif"] = "SimHei"  # 设置中文字体
plt.violinplot(dataset=data, showmeans=True, showmedians=False, points=1000, vert=True)

# 设置刻度字体大小
plt.xticks([1, 2, 3], ["苹果酸", "镁", "脯氨酸"], fontsize=15)
plt.yticks(fontsize=15)
plt.title("葡萄酒成分含量小提琴图", fontsize=15)
plt.show()
